package prastice;



import java.util.HashMap;

public class CopyRandomList {
    /**
     * map 可以使
     * 旧节点的 next 所对应的值
     * 与新节点 next 相匹配
     */
    public Node copyRandomList(Node head) {
        HashMap<Node, Node> map = new HashMap<>();
        // 1. 先遍历一次 存储 原先链表与新链表的节点
        Node cur = head;
        while(cur != null) {
            map.put(cur, new Node(cur.val));
            cur = cur.next;
        }

        // 2. 再次遍历 修改新链表所对应的 next 与 random
        cur = head;
        while(cur != null) {
            Node newNode = map.get(cur); // 新节点
            newNode.next = map.get(cur.next);
            newNode.random = map.get(cur.random);
            cur = cur.next;
        }

        return map.get(head);
    }
}
